package leo.mystudy.leetcode;

import leo.mystudy.pojo.TreeNode;

import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author leo
 * @version 1.0.0
 * @description 给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。
 * <p>
 *  
 * <p>
 * 示例 1:
 * <p>
 * <p>
 * <p>
 * 输入: [1,2,3,null,5,null,4]
 * 输出: [1,3,4]
 * 示例 2:
 * <p>
 * 输入: [1,null,3]
 * 输出: [1,3]
 * 示例 3:
 * <p>
 * 输入: []
 * 输出: []
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/binary-tree-right-side-view
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @create 2022/3/12 20:54
 */

public class No199 {

    public List<Integer> rightSideView(TreeNode root) {
        Map<Integer, Integer> map = new HashMap<>();
        preForeachForRightView(root, 0, map);
        return map.entrySet().stream().sorted((Map.Entry.comparingByKey())).map(Map.Entry::getValue).collect(Collectors.toList());
    }

    public void preForeachForRightView(TreeNode treeNode, int dep, Map<Integer, Integer> map) {
        if (treeNode == null) {
            return;
        }
        map.put(dep, treeNode.val);
        preForeachForRightView(treeNode.left, dep + 1, map);
        preForeachForRightView(treeNode.right, dep + 1, map);
    }
}
